package com.xiao.xin.core.dto;

import lombok.Data;

@Data
public class Text2ImageDto {
    private int size;
    private int model;
    private int scale = 1;
    private int step;
    private int cfg;
    private int sampler;
    private int seed;
    private String reverse;
    private String propmt;
    private String clientId;

    public String modelName(){
        if (model == 1){
            return "majicmixRealistic_v7.safetensors";
        }else {
            return "anythingelseV4_v45.safetensors";
        }
    }

    public int width() {
        if(scale==3){
            return 768;
        } else {
            return 512;
        }
    }

    public int height() {
        if(scale==2){
            return 768;
        } else {
            return 512;
        }
    }

    public String samplerName(){
        switch (sampler){
            case 1:
                return "dpmpp_sde";
            case 2:
                return "dpmpp_2m";
            case 3:
                return "euler";
            case 4:
                return "dpmpp_3m_sde";
            default:
                return "euler";
        }
    }

    public String scheduler(){
        return "karras";
    }



}
